import onnx
import onnx.helper as helper

model = onnx.load('../tensorrt_plugin/gemv.onnx')
for item in model.graph.node:
    if(item.name == 'MatMul_0'):
        item.CopyFrom(helper.make_node(op_type = "Sxai",
                                         inputs= item.input,
                                         outputs = item.output,
                                         name = 'Sxai_0'))
onnx.save(model, 'gemv.onnx')